package com.yc.snack.entity;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

import javax.imageio.ImageIO;

/**
 * 图片水印工具类
 * @author lenovo
 *
 */
public class WaterMarkUtil {
	private static final String FONT_NAME = "宋体";
	private static final int FONT_STYLE = Font.BOLD; //粗体
	private static final int FONT_SIZE = 20; // 字体大小
	private static final Color FONT_COLOR = Color.RED; //字体颜色
	private static final String MARK_TEXT = "源辰信息"; //水印文字

	/**
	 * 给指定的图片右下角加上水印文字
	 * 等用户将图片上传到服务器之后再水印
	 * @param pic
	 */
	public static void textWaterMark(File pic) {
		FileOutputStream fos = null;
		try {
			//1、获取要添加的水印图片
			Image image = ImageIO.read(pic);
			
			// 2、获取图片大小
			int width = image.getWidth(null);
			int height = image.getHeight(null);
			
			//3、创建图片缓冲对象
			BufferedImage bufferedImage = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
			
			//4、获取缓冲区图片的绘制工具
			Graphics2D g = bufferedImage.createGraphics();
			
			
			//5、绘制原图片
			g.drawImage(image,0,0,width,height,null);
			
			//6、在绘制水印
			//（1） 绘制水印文本字体、样式
			g.setFont(new Font(FONT_NAME, FONT_STYLE,FONT_SIZE));
			
			//(2) 设置字体颜色
			g.setColor(FONT_COLOR);
			
			//(3)社会文本大小，以便定位
			int x = width = getTextWidth(MARK_TEXT) * FONT_SIZE - 10;
			int y = height - FONT_SIZE;//这个水印文本距离顶部的距离为 图片的高度-字体大小
			//(4) 设置透明度
			g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,0.8F));
			
			//(5)设置文本位置
			g.drawString(MARK_TEXT, x, y);
			
			//7 、解放工具
			g.dispose();
			
			fos = new FileOutputStream(pic.getAbsolutePath());
			
			//8、将图片缓冲区的数据写入到文件中
			ImageIO.write(bufferedImage,"JPEG",fos);
		}catch(IOException e) {
			e.printStackTrace();
		}finally {
			if(fos != null) {
				try {
					fos.close();
				}catch(IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * 获取文本的宽度
	 * @param text
	 * @return
	 */
	private static int getTextWidth(String text) {
		int length = text.length(); // 获取水印字符串的长度
		String temp = null;
		//循环获取每一个字符
		for(int i = 0,len = text.length(); i < len; ++i) {
			temp = String.valueOf(text.charAt(i)); //获取当前循环得到的字符
			if(temp.getBytes().length>1) { //说明这个不是一个英文字符
				++length;
			}
		}
		length = length % 2 == 0 ? length / 2 :length /2 + 1;
		
		return length;
	}
}
